package com.huawei.controller;

import java.util.Stack;

/**
 * 394. 字符串解码
 *
 * @author cwx1134046
 * @since 2023-03-01
 */
public class DecodeString {
    /**
     * 给定一个经过编码的字符串，返回它解码后的字符串。
     *
     * 编码规则为: k[encoded_string]，表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。
     *
     * 你可以认为输入字符串总是有效的；输入字符串中没有额外的空格，且输入的方括号总是符合格式要求的。
     *
     * 此外，你可以认为原始数据不包含数字，所有的数字只表示重复的次数 k ，例如不会出现像 3a 或 2[4] 的输入。
     *
     *  
     *
     * 示例 1：
     *
     * 输入：s = "3[a]2[bc]"
     * 输出："aaabcbc"
     * 示例 2：
     *
     * 输入：s = "3[a2[c]]"
     * 输出："accaccacc"
     * 示例 3：
     *
     * 输入：s = "2[abc]3[cd]ef"
     * 输出："abcabccdcdcdef"
     * 示例 4：
     *
     * 输入：s = "abc3[cd]xyz"
     * 输出："abccdcdcdxyz"
     *
     */

    /**
     *
     */
    public static String decodeString(String expression) {
        int oldType = -1;
        Stack<Node> stack = new Stack<>();
        StringBuilder res = new StringBuilder();

        for (int i = 0; i< expression.length(); i++) {
            char c = expression.charAt(i);
            int type = getType(c);

            //开始
            if(oldType == -1){
                if(type == 0){
                    Node node = new Node();
                    node.num = Integer.valueOf(""+c);
                    stack.push(node);
                } else if(type == 1){
                    res.append(c);
                }
            }
            else if(oldType == 0){
                //上一个数据类型为 数字
                // 上一个数据类型为数字，且下一个还是数字
                if(type == 0){
                    Node peek = stack.peek();
                    peek.num = Integer.valueOf(peek.num + ""+c);
                }
            }else if(oldType == 2){
                //上个数据类型为[,现在为字母
                if(type == 1) {
                    Node peek = stack.peek();
                    peek.str.append(c);
                }
                // 3[3[a]b]
                if(type == 0) {
                    Node node = new Node();
                    node.num = Integer.valueOf(""+c);
                    stack.push(node);
                }
            }else if(oldType == 3){
                //上个数据类型为]
                if(type == 3){
                    Node pop = stack.pop();
                    if(stack.isEmpty()){
                        res.append(pop.deal());
                    }else {
                        stack.peek().str.append(pop.deal());
                    }
                }
                //文本
                if(type == 1){
                    if(stack.isEmpty()){
                        res.append(c);
                    }else {
                        stack.peek().str.append(c);
                    }
                }
                //数字
                if(type == 0){
                    Node node = new Node();
                    node.num = Integer.valueOf(""+c);
                    stack.push(node);
                }
            }else if(oldType == 1){
                //上个数据类型是 文本
                if(type == 1){
                    if(!stack.isEmpty()){
                        stack.peek().str.append(c);
                    }else{
                        res.append(c);
                    }
                }else if (type == 3){
                    Node pop = stack.pop();
                    if(stack.isEmpty()){
                        res.append(pop.deal());
                    }else {
                        stack.peek().str.append(pop.deal());
                    }
                }else if(type == 0){
                    Node node = new Node();
                    node.num = Integer.valueOf(""+c);
                    stack.push(node);
                }

            }

            oldType = type;
        }

        return res.toString();
    }

    public static void main(String[] args) {
        String s = decodeString("100[leetcode]");
        System.out.println(s);
    }

    /**
     * 0 数字
     * 1 a-z
     * 2 [
     * 3 ]
     */
    static int getType(Character ch) {
        if(ch == '['){
            return 2;
        }
        if(ch == ']') {
            return 3;
        }
        if('0' <= ch  && ch <= '9'){
            return 0;
        }
        return 1;

    }

    static class Node{
        int num;
        StringBuilder str = new StringBuilder();

        String deal(){
            StringBuilder res = new StringBuilder();
            for (int i = 0 ;i < num ; i++){
                res.append(str);
            }
            return res.toString();
        }
    }

}
